python code for fractal tree